package primetest.faces;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;


import primetest.dto.WardDto;

public class WardListModel extends ListDataModel<WardDto> implements SelectableDataModel<WardDto> {

	public WardListModel(List<WardDto> wardList){
		super(wardList);
	}

	@Override
	public WardDto getRowData(String rowKey) {
		@SuppressWarnings("unchecked")
		List<WardDto> wardList = (List<WardDto>)getWrappedData();
		for (WardDto wardDto : wardList) {
			if(wardDto.getKey() == rowKey){
				return wardDto;
			}
		}
		return null;
	}

	@Override
	public Object getRowKey(WardDto wardDto) {
		return wardDto.getKey();
	}

}
